// Package dsmysql 提供MySQL客户端的封装和工具函数
package dsmysql

import (
	"fmt"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

// Client MySQL客户端封装
type Client struct {
	*gorm.DB
}

// NewClient 创建新的MySQL客户端
func NewClient(config Config) (*Client, error) {
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local",
		config.User,
		config.Password,
		config.Host,
		config.Port,
		config.Database,
		config.Charset,
	)

	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		return nil, fmt.Errorf("连接MySQL失败: %v", err)
	}

	return &Client{
		DB: db,
	}, nil
}

// Close 关闭MySQL连接
func (c *Client) Close() error {
	sqlDB, err := c.DB.DB()
	if err != nil {
		return err
	}
	return sqlDB.Close()
}
